package pl.skubicha.service.dto.util;

import java.util.Set;

import pl.skubicha.domain.CreditingForm;
import pl.skubicha.domain.FinalGrades;
import pl.skubicha.domain.PartialGrades;
import pl.skubicha.response.CreditingFormDTO;

public class CreditingFormTestUtil {

	public static CreditingFormDTO createDTO( final Integer id, final String name, final Set< PartialGrades > partialGrades, final Set< FinalGrades > finalGrades ) {
		final CreditingFormDTO dto = new CreditingFormDTO();
		dto.setId( id );
		dto.setName( name );
		dto.setPartialGrades( partialGrades );
		dto.setFinalGrades( finalGrades );

		return dto;
	}

	public static CreditingForm createModelObject( final Integer id, final String name ) {
		final CreditingForm model = new CreditingForm( name );
		model.setId( id );

		return model;
	}

	public static CreditingForm createModelObject( final Integer id, final String name, final Set< PartialGrades > partialGrades ) {
		final CreditingForm model = new CreditingForm( name );
		model.setId( id );
		model.setPartialGrades( partialGrades );

		return model;
	}
}